package test230601;

/**
 * @author 兴趣使然黄小黄
 * @version 1.0
 * @date 2023/6/1 19:40
 * 188. 买卖股票的最佳时机IV
 * https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-iv/
 */
public class Solution05 {

    private final int INF = 0x3f3f3f3f;

    public int maxProfit(int k, int[] prices) {
        int n = prices.length;  // 天数
        int[][] f = new int[n][k+1];  // f[i][j] 表示第 i 天处于"买入"状态的最大利润, j 表示交易次数
        int[][] g = new int[n][k+1];  // g[i][j] 表示第 i 天处于"卖出"状态的最大利润, j 表示交易次数
        // 状态初始化, 一开始交易次数需要足够小
        for (int i = 0; i <= k; ++i) {
            f[0][i] = g[0][i] = -INF;
        }
        f[0][0] = -prices[0];  // 第一天买入后才能处于买入状态
        g[0][0] = 0;
        // 根据状态机进行迭代过程
        for (int i = 1; i < n; ++i) {
            for (int j = 0; j <= k; ++j) {
                f[i][j] = Math.max(f[i-1][j], g[i-1][j] - prices[i]);
                if (j - 1 >= 0) {
                    g[i][j] = Math.max(g[i-1][j], f[i-1][j-1] + prices[i]);
                } else {
                    g[i][j] = g[i-1][j];
                }
            }
        }
        // 找到最大利润
        int maxProfit = Integer.MIN_VALUE;
        for (int i = 0; i <= k; i++) {
            maxProfit = Math.max(g[n-1][i], maxProfit);
        }
        return maxProfit;
    }
}
